package db

import (
	"awesomeProject2/model"
	"gorm.io/driver/mysql"
	"gorm.io/gorm"
	"log"
)

var DB *gorm.DB

func InitDB(dsn string) error {
	var err error

	// 打开 MySQL 连接（GORM）
	DB, err = gorm.Open(mysql.Open(dsn), &gorm.Config{})
	if err != nil {
		log.Println("❌ 数据库连接失败:", err)
		return err
	}

	// 自动迁移建表
	if err := DB.AutoMigrate(&model.Wallet{}, &model.Order{}, &model.Trade{}); err != nil {
		log.Println("❌ 自动建表失败:", err)
		return err
	}

	log.Println("✅ MySQL 初始化完成，表已创建/更新")
	return nil
}
